import os
import xlrd2

"""
    unittest中 参数化 从文件 或者 excel中来 读取 其中的数据  io
"""

"""
    编写一个方法来测试 文件是否存在
"""


def file_exist(file_path):
    if os.path.isfile(file_path):
        return True
    raise Exception('文件不存在')


"""
    编写一个方法来 读取指定文件中的数据
"""


def read_text(file_path, seq=','):
    file_exist(file_path)
    with open(file=file_path, mode='r', encoding='utf-8') as f:
        result = []
        for line in f:
            result.append(tuple(line.strip().split(seq)))
        return result


"""
    编写一个方法来 读取指定文件(xls、xlsx)中的数据
"""


def read_excel(file_path):
    file_exist(file_path)
    book = xlrd2.open_workbook(file_path)
    sheet = book.sheet_by_index(0)

    result = []
    for i in range(sheet.nrows):
        result.append(tuple(sheet.row_values(i)))
    return result


if __name__ == '__main__':
    print(read_excel('data.xlsx'))
